version = 'case_v0'
description = 'to show the mind is right or not...'
# 模型相关

BACKBONE_NET = 'darknet53'
IS_CONF_OUT_ADD_SIGMOID = True
IS_XY_OUT_ADD_SIGMOID = True

CONF_LOSS_TYPE = 'mse'
BOXREGRESSOIN_LOSS_TYPE = 'mse'
CLASSFICATION_LOSS_TYPE = 'mse'

OBJ_LOSS_WEIGHT = 1
NOOBJ_LOSS_WEIGHT = 1

#训练相关
OPTMIZER = 'adam'
LR = 0.003
LR_STEP= 10
LOG_OUT = fr"./{version}_log.log"
BATCH_SIZE = 24


# FREEZE_BONE_NET = True
FREEZE_BONE_NET = False
WEIGHTS_TO_LOAD = 'darknet53.conv.74'
# WEIGHTS_TO_LOAD = r'weights/case_v0_best.pth'
WEIGHTS_SAVE_PATH = fr'weights/{version}'

END_EPOCH=1000


# 数据相关,更新数据集时需要更新这部分
DATA_FROM = 'VOC2012'
classes = [
    "aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat",
    "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person",
    "pottedplant", "sheep", "sofa", "train", "tvmonitor"
]
class_number = 20
img_width = 416
anchor_box = {
    13:[[15.0, 22.0],[45.0, 31.0],[30.0, 62.0]],
    26:[[90.0, 70.0],[55.0, 119.0],[100.0, 174.0]],
    52:[[228.0, 129.0],[165.0, 243.0],[312.0, 275.0]]
}